import pymongo
from docx import Document


def write_word():
    # 连接服务器
    client = pymongo.MongoClient('localhost', 27017)

    # 获取要操作的数据库
    db = client['baiduNovel']

    # 获取要操作的集合
    collection = db['baiduFanRen']

    # 查询全部数据
    result = collection.find()

    # 创建Word文档并写入内容
    doc = Document()

    # 写入每一章的内容
    for item in result:
        doc.add_heading(item['title'], level=1)  # 将标题作为一级标题插入
        doc.add_paragraph(item['content'])  # 插入内容段落（自动换行）
        doc.add_page_break()  # 每章结束后添加分页（可选，让章节分页显示）

    # 保存文档，文件名使用小说名
    # 处理文件名中的特殊字符（防止保存失败）
    safe_filename = "凡人修仙传.docx"
    doc.save(safe_filename)
    print(f"已生成文档：{safe_filename}")